#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
数据收集员 Agent
负责收集股票数据
"""

from crewai import Agent
from prompts import get_data_collector_prompt


def create_data_collector_agent(tools: list, llm=None) -> Agent:
    """
    创建数据收集员智能体
    
    Args:
        tools: 可用的工具列表（必需）
        llm: 语言模型实例（可选）
    
    Returns:
        配置好的数据收集员 Agent
    """
    agent_config = {
        "role": "股票数据收集专家",
        "goal": "使用工具收集股票的实时数据、历史数据和市场新闻",
        "backstory": get_data_collector_prompt(),
        "tools": tools,
        "verbose": True,
        "allow_delegation": False
    }
    
    if llm is not None:
        agent_config["llm"] = llm
    
    return Agent(**agent_config)
